package com.xzr.demo.mqserver.datacenter;

import com.xzr.demo.DemoApplication;
import com.xzr.demo.mqserver.core.Binding;
import com.xzr.demo.mqserver.core.Exchange;
import com.xzr.demo.mqserver.core.ExchangeType;
import com.xzr.demo.mqserver.core.MSGQueue;
import com.xzr.demo.mqserver.mapper.MetaMapper;

import java.io.File;
import java.util.List;

/**
 * 通过这个类， 整洁 数据库 操作
 * @Author：徐志荣
 * @Date：2023/8/8 上午 9:48
 */
public class DataBaseManager {

    private MetaMapper metaMapper;

    /**
    *针对数据库进行初始化
     * 建库建表 + 插入默认数据
    * @param
    * @return
    */
    public void init(){
        // 手动获取到 MetaMapper
        metaMapper = DemoApplication.context.getBean(MetaMapper.class);
        if(!checkDBExists()){
            //数据库不存在
            //先创建 data 目录
            File dataDir = new File("./data");
            dataDir.mkdirs();

            createTable();
            createDefaultData();
            System.out.println("[DataBaseManager] 数据库初始化完成！");
        } else {
            //数据库存在
            System.out.println("[DataBaseManager] 数据库已经存在！");

        }
    }




    private boolean checkDBExists() {
        File file = new File("./data/meta.db");
        if(file.exists()){
            return true;
        }
        return  false;
    }
    /**
    * 删除数据库
    *
    * @return
    */
    public void  deleteDB() {
        File file = new File("./data/meta.db");
        boolean ret =  file.delete();
        if(ret){
            System.out.println("[DataBaseManager] 删除数据库成功！");
        }else{
            System.out.println("[DataBaseManager] 删除数据库失败！");

        }
        File dataDir = new File("./data");
        //delete 删除目录 前提： 目录下是空的
       ret =  dataDir.delete();
        if(ret){
            System.out.println("[DataBaseManager] 删除数据库目录成功！");
        }else{
            System.out.println("[DataBaseManager] 删除数据库目录失败！");

        }
    }


    /**
    * 这个方法 用来建表 ， （首次执行，Mybatis自动创建 meta.db）
    *
    * @return
    */
    private void createTable() {
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        System.out.println("[DataBaseManager] 创建表完成！");

    }

    /**
    * 给数据库表中，添加默认的数据
     * 此处，添加默认的交换机
     * RabbitMQ 里面有一个这样的设定：带有一个 匿名 的交换机， 类型是 DIRECT
    *
    * @return
    */
    private void createDefaultData() {
        //构造一个默认的交换机
        Exchange exchange = new Exchange();
        exchange.setName("");
        exchange.setType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        metaMapper.insertExchange(exchange);
        System.out.println("[DataBaseManager] 初始化数据完成！");

    }


    //把其他的数据库的操作，也在这个类中封装一下
    public void insertExchange(Exchange exchange){
        metaMapper.insertExchange(exchange);
    }

    public  void deleteExchange(String exchangeName){
        metaMapper.deleteExchange(exchangeName);
    }
    public List<Exchange> selectAllExchanges(){
       return metaMapper.selectAllExchanges();
    }
    public void insertQueue(MSGQueue queue){
        metaMapper.insertQueue(queue);
    }
    public void deleteQueue(String queueName){
        metaMapper.deleteQueue(queueName);
    }
    public List<MSGQueue> selectAllQueues(){
        return   metaMapper.selectAllQueues();
    }
    public void insertBinding(Binding binding){
        metaMapper.insertBinding(binding);
    }
    public void deleteBinding(Binding binding){
        metaMapper.deleteBinding(binding);
    }
    public List<Binding> selectAllBindings(){
        return  metaMapper.selectAllBindings();
    }
}
